package com.bsj.power.common.util.excel;

import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;

import java.util.HashMap;
import java.util.Map;

/**
 * @description: excel文件通用解析
 * @author: LiJiaXin
 * @time: 2022/5/9 10:23
 */
@Data
@Slf4j
public class ModelExcelListener<T> extends AnalysisEventListener {

    /**
     * 自定义用于暂时存储data
     * 可以通过实例获取该值
     */
    private Map<Integer, T> datas = new HashMap<Integer, T>();

    /**
     * 每解析一行都会回调invoke()方法
     *
     * @param object  读取后的数据对象
     * @param context 内容
     */
    @Override
    public void invoke(Object object, AnalysisContext context) {
        final T t = (T) object;
        Integer rowIndex = context.readRowHolder().getRowIndex() + 1;
        //数据存储到map，供批量处理，或后续自己业务逻辑处理。
        datas.put(rowIndex, t);
        //根据自己业务做处理
    }

    @Override
    public void doAfterAllAnalysed(AnalysisContext context) {
        log.info("excel parse success !!!");
    }
}
